file(GLOB NVLINK_SOURCES "*.cpp")

if (USE_HIP)
  hipify_files(NVLINK_SOURCES)
endif()

add_library(nvlink_transport OBJECT ${NVLINK_SOURCES})

if (USE_CUDA)
target_include_directories(nvlink_transport PUBLIC  CUDA::cudart "/usr/local/cuda/include")
endif()

if (USE_HIP)
target_include_directories(nvlink_transport PUBLIC ${HIP_INCLUDE_DIRS})
endif()
